rm -rf ./trans_results
mkdir trans_results

# main process
echo "############################################"
echo "Convert pytorch pt model to keras h5 model begin"
echo "############################################"
python pytorch_to_keras.py --input_model="./UltraNet_4w4a.pt" --output_model="./trans_results/UltraNet_4w4a.h5"
python pytorch_to_keras.py --input_model="./UltraNet_float.pt" --output_model="./trans_results/UltraNet_float.h5"
echo "############################################"
echo "Convert completed"
echo "############################################"